# -*- coding: utf-8 -*-
from odoo import http
from odoo.http import request
import json
import base64


class ItemMasterImportController(http.Controller):
    
    @http.route('/production_planning/import_csv', type='http', auth='user', methods=['POST'], csrf=False)
    def import_csv(self, **kwargs):
        """处理CSV文件导入"""
        try:
            # 获取上传的文件
            file_data = kwargs.get('file')
            if not file_data:
                return json.dumps({
                    'success': False,
                    'message': '请选择要导入的文件'
                })
            
            # 调用模型方法导入数据
            result = request.env['item.master'].import_csv_data(file_data)
            
            return json.dumps(result)
            
        except Exception as e:
            return json.dumps({
                'success': False,
                'message': f'导入失败：{str(e)}'
            })
    
    @http.route('/production_planning/import_excel', type='http', auth='user', methods=['POST'], csrf=False)
    def import_excel(self, **kwargs):
        """处理Excel文件导入"""
        try:
            # 获取上传的文件
            file_data = kwargs.get('file')
            if not file_data:
                return json.dumps({
                    'success': False,
                    'message': '请选择要导入的文件'
                })
            
            # 调用模型方法导入数据
            result = request.env['item.master'].import_excel_data(file_data)
            
            return json.dumps(result)
            
        except Exception as e:
            return json.dumps({
                'success': False,
                'message': f'导入失败：{str(e)}'
            })
    
    @http.route('/production_planning/export_template_csv', type='http', auth='user')
    def export_template_csv(self):
        """导出CSV模板"""
        return request.env['item.master'].export_template_csv()
    
    @http.route('/production_planning/export_template_excel', type='http', auth='user')
    def export_template_excel(self):
        """导出Excel模板"""
        return request.env['item.master'].export_template_excel()

